//
//  Extention+UIImage.swift
//  WeiboSwift
//
//  Created by 战场原黑仪 on 16/7/29.
//  Copyright © 2016年 战场原黑仪. All rights reserved.
//

import UIKit

extension UIImage {
    
    //获取截屏结果
    static func getScreenShot() -> UIImage {
        
        //获取视图
        let window = UIApplication.sharedApplication().keyWindow!
        
        //开启图片上下文
        UIGraphicsBeginImageContext(window.bounds.size)
        
        //将当前界面渲染到图片上下文中
        window.drawViewHierarchyInRect(window.bounds, afterScreenUpdates: false)
        
        //从图片上下文中获取图片
        let image = UIGraphicsGetImageFromCurrentImageContext()
        
        //关闭图片上下文
        UIGraphicsEndImageContext()
        
        //返回图片
        return image
    }
    
    
    //对图片进行压缩处理
    //压成指定宽度
    func getImageScale(width: CGFloat) -> UIImage {
        
        //判断当前图片宽度是否小于指定压缩宽度
        if self.size.width < width {
            return self
        }
        
        //处理得出压缩后的图片高度
        let height = (self.size.height / self.size.width) * width
        
        //开启图片上下文
        UIGraphicsBeginImageContext(CGSize(width: width, height: height))
        //将图片渲染到上下文中
        self.drawInRect(CGRect(x: 0, y: 0, width: width, height: height))
        //从上下文中获取图片
        let newImage = UIGraphicsGetImageFromCurrentImageContext()
        //关闭上下文
        UIGraphicsEndImageContext()
        //返回图片
        return newImage
    }
}
